import time
import pymysql
import streamlit as st

def init_meaaage():
    st.set_page_config(layout="wide",
                       page_icon="😀",
                       page_title="聊天机器人")
    st.title("聊天框")
    return 0

def get_connection():
    conn = pymysql.connect(
        host='localhost',
        port=3306,
        user='root',
        passwd='root',
        db='jupyter',
        charset='utf8',
        cursorclass=pymysql.cursors.DictCursor
    )
    return conn

def chat_response():
    message="大学时期我学习电子学。在当时，电子学处在一个非常令人兴奋的时期，电脑已经出现，但还没有进入家庭，仍然属于工业化的产物。我去了一所有导师制度的大学。很幸运，我的导师 Richard Grimsdale（理查德·格里姆斯代尔）是建造世界上第一台晶体管计算机的人。当时，我并不知道这很重要。他的导师又是艾伦·图灵，被誉为英国计算机之父和人工智能之父。在那里的课程中，我学习了编程。当我离开大学时，我寻找了一份编程工作。事实上，有一段时间我得到了一份非常微不足道的工作，只需简单输入研究机构或研究中心的数据。但由于会编程，我还被安排做一些编程任务，我做得很投入，也很喜欢。这样工作三个月后，公司雇用我成为了一名正式的程序员。这是一个很棒的环境，让我有很多自由的空间可以去学习。我当时正在为研究项目进行编程，这本身就非常有趣。我热衷于学习计算领域的新事物。当时我是在英国，那是我长大的地方。巧合的是，我当时工作的信息来源之一是荷兰的这个研究所，也就是我现在工作的地方。我很早就了解了CWI，发现他们所做的事情非常令人兴奋。我花了很多年时间，最终来到了CWI。"
    message.encode('utf-8')
    for word in message.split():
        yield word+" "
        time.sleep(1)
def main():
    init_meaaage()
    st.title("欢迎使用有问必答chatBot")
    message=st.chat_input("请输入你的问题")
    if message:
        with st.chat_message("YiHe"):
            st.write(message)
        with st.chat_message("Robot"):
            st.image("http://gips0.baidu.com/it/u=1690853528,2506870245&fm=3028&app=3028&f=JPEG&fmt=auto?w=1024&h=1024",
                     width=100)
            message=chat_response()
            st.write_stream(message)
if __name__ == '__main__':
    main()